package com.jse.util;

import java.awt.image.BufferedImage;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.InputStream;
import java.net.HttpURLConnection;
import java.net.MalformedURLException;
import java.net.URISyntaxException;
import java.net.URL;
import java.util.UUID;

import org.nutz.img.Images;
import org.nutz.mvc.Mvcs;

public class ImageUtil {

	public static String save(String urlstr) throws Exception {
		// new一个URL对象
		URL url = new URL(urlstr);
		// 打开链接
		HttpURLConnection conn = (HttpURLConnection) url.openConnection();
		// 设置请求方式为"GET"
		conn.setRequestMethod("GET");
		// 超时响应时间为5秒
		conn.setConnectTimeout(5 * 1000);
		// 通过输入流获取图片数据
		InputStream inStream = conn.getInputStream();
		// 得到图片的二进制数据，以二进制封装得到数据，具有通用性
		byte[] data = readInputStream(inStream);
		// new一个文件对象用来保存图片，默认保存当前工程根目录
		String filestr = "upload/images/" + UUID.randomUUID().toString() + ".png";
		File imageFile = new File(filestr);
		// 创建输出流
		String str = Mvcs.getServletContext().getRealPath("/").replace("\\", "/");// web中才可用
		String appName = str.substring(0, str.lastIndexOf("/"));
		appName = appName.substring(appName.lastIndexOf("/"));
		// 保存在tomcat/webapps/appName/temp/img目录下
		String filePath = str.substring(0, str.indexOf("tomcat") + 7) + "webapps/upload" + appName + "/temp/img/";
		FileOutputStream outStream = new FileOutputStream(filePath + imageFile);
		// 写入数据
		outStream.write(data);
		// 关闭输出流
		outStream.close();
		return filestr;
	}

	public static byte[] readInputStream(InputStream inStream) throws Exception {
		ByteArrayOutputStream outStream = new ByteArrayOutputStream();
		// 创建一个Buffer字符串
		byte[] buffer = new byte[1024];
		// 每次读取的字符串长度，如果为-1，代表全部读取完毕
		int len = 0;
		// 使用一个输入流从buffer里把数据读取出来
		while ((len = inStream.read(buffer)) != -1) {
			// 用输出流往buffer里写入数据，中间参数代表从哪个位置开始读，len代表读取的长度
			outStream.write(buffer, 0, len);
		}
		// 关闭输入流
		inStream.close();
		// 把outStream里的数据写入内存
		return outStream.toByteArray();
	}

	public static void main(String[] args) throws MalformedURLException, URISyntaxException {
		BufferedImage image = Images.read(new URL("https://www.baidu.com/img/bdlogo.png"));
		Images.write(image, new File("d:/abc.png"));
	}
}
